package cn.iocoder.yudao.module.aiposter.remote.vo;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;

@Data
@AllArgsConstructor
@NoArgsConstructor
@ToString
public class SegmentationParam {
    /**
     * 默认为默认为“general-s”；
     * 其他选项：
     * "general-1": 通用抠图1、
     * "general-2": 通用抠图2、
     * "general-s": 通用抠图-小模型、
     * "human": 人物/人像抠图、
     * "cloth": 服装/衣服抠图、
     * "anime": 动漫图片抠图",
     */
    private String segModel = "general-s";
    /**
     * 默认为“region”；
     * 其他选项：
     * "region": 输出抠图区域,原图中其他背景区域被丢弃;
     * "all": 输出完整区域抠图区域，尺寸与原图一致，其他都为背景",
     */
    private String segPattern = "region";
    /**
     * 默认为“alpha”图层;
     * 其他选项：
     * "alpha": 图层,返回RGBA图像，背景为透明，可以直接贴图；
     * "mask": 黑白（二值）图,抠图（前景）区域为白色（1），背景为（0）"
     */
    private String returnType = "alpha";
}
